package com.xxxx.server.controller;

import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.xxxx.server.pojo.Menu;
import com.xxxx.server.pojo.MenuRole;
import com.xxxx.server.pojo.RespBean;
import com.xxxx.server.pojo.Role;
import com.xxxx.server.service.IMenuRoleService;
import com.xxxx.server.service.IMenuService;
import com.xxxx.server.service.IRoleService;
import io.swagger.annotations.ApiOperation;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.DeleteMapping;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.PathVariable;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.PutMapping;
import org.springframework.web.bind.annotation.RequestBody;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;

import java.util.List;
import java.util.stream.Collectors;

/**
 * 权限组
 *
 * @author zhoubin
 * @since 1.0.0
 */
@RestController
@RequestMapping("/system/basic/permiss")
public class PermissController {

	@Autowired
	private IRoleService roleService;
	@Autowired
	private IMenuService menuService;
	@Autowired
	private IMenuRoleService menuRoleService;


	@ApiOperation(value = "获取所有角色")
	@GetMapping("/")
	public List<Role> getAllRoles(){
		return roleService.list();
	}

	@ApiOperation(value = "添加角色")
	@PostMapping("/role")
	public RespBean addRole(@RequestBody Role role){
		if (!role.getName().startsWith("ROLE_")){
			role.setName("ROLE_"+role.getName());
		}
		if (roleService.save(role)){
			return RespBean.success("添加成功!");
		}
		return RespBean.error("添加失败!");
	}

	@ApiOperation(value = "删除角色")
	@DeleteMapping("/role/{rid}")
	public RespBean deleteRole(@PathVariable Integer rid){
		if (roleService.removeById(rid)){
			return RespBean.success("删除成功！");
		}
		return RespBean.error("删除失败!");
	}

	@ApiOperation(value = "查询所有菜单")
	@GetMapping("/menus")
	public List<Menu> getAllMenus(){
		return menuService.getAllMenus();
	}

	@ApiOperation(value = "根据角色id查询菜单id")
	@GetMapping("/mid/{rid}")
	public List<Integer> getMidByRid(@PathVariable Integer rid){
		return menuRoleService.list(new QueryWrapper<MenuRole>().eq("rid",rid)).stream().map(MenuRole::getMid).collect(Collectors.toList());
	}

	@ApiOperation(value = "更新角色菜单")
	@PutMapping("/")
	public RespBean updateMenuRole(Integer rid,Integer[] mids){
		return menuRoleService.updateMenuRole(rid,mids);
	}
}